package com.javaee.exam.split;

import java.util.Locale;
import java.util.Properties;

import org.apache.ibatis.executor.Executor;
import org.apache.ibatis.executor.keygen.SelectKeyGenerator;
import org.apache.ibatis.mapping.BoundSql;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.plugin.Interceptor;
import org.apache.ibatis.plugin.Intercepts;
import org.apache.ibatis.plugin.Invocation;
import org.apache.ibatis.plugin.Plugin;
import org.apache.ibatis.plugin.Signature;
import org.apache.ibatis.session.ResultHandler;
import org.apache.ibatis.session.RowBounds;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.transaction.support.TransactionSynchronizationManager;

/**
 * @author Administrator
 * @desc 设置MyBatis 的拦截器(需要使用DataSource 路由就需要通过MyBatis 拦截器，拦截SQL 请求，根据SQL
 *       类型选择不同的数据源) DynamicDataSourceInterceptor 拦截器写好没用需要在 MyBatis 配置文件中配置方可使用
 *       <plugins> 
 *       	<plugin interceptor="DynamicDataSourceInterceptor拦截器类路径" />
 *       </plugins>
 */
@Intercepts({
	@Signature(type=Executor.class,method="update",args= {MappedStatement.class,Object.class}),
	@Signature(type=Executor.class,method="query",args= {MappedStatement.class,Object.class,RowBounds.class,ResultHandler.class}),
})
public class DynamicDataSourceInterceptor implements Interceptor {
	Logger logger = LoggerFactory.getLogger(DynamicDataSourceInterceptor.class);

	// u0020 表示空格
	private static final String REGEX = ".*insert\\u0020.*|.*delete\\u0020.*|.*update\\u0020.*";

	/*
	 * intercept 是决定使用那个数据源
	 */
	@Override
	public Object intercept(Invocation invocation) throws Throwable {
		// 判断是不是事务。true 是有事务
		boolean synchronizationActive = TransactionSynchronizationManager.isActualTransactionActive();
		String lookupkey = DynamicDataSourceHolder.DB_MASTET;// 决定DataSource
		// 获取CRUD 的参述
		Object[] objects = invocation.getArgs();
		MappedStatement ms = (MappedStatement) objects[0];

		if (synchronizationActive != true) {
			// 读操作
			if (ms.getSqlCommandType().equals(SqlCommandType.SELECT)) {
				// selectKey 为自增ID 查询主键(select LAST_INSERT_ID())方法，使用主库
				if (ms.getId().contains(SelectKeyGenerator.SELECT_KEY_SUFFIX)) {
					lookupkey = DynamicDataSourceHolder.DB_MASTET;
				} else {
					// objects 第二个参述就是 SQL
					BoundSql boundSql = ms.getSqlSource().getBoundSql(objects[1]);
					// 根据中国时区，将所有制表符，换行符替换成空格
					String sql = boundSql.getSql().toLowerCase(Locale.CHINA).replaceAll("[\\t\\n\\r]", " ");
					// 正则 判断是否是insert,delete,update 开头，是使用主库，不是使用从库
					if (sql.matches(REGEX)) {
						// 写
						lookupkey = DynamicDataSourceHolder.DB_MASTET;
					} else {
						// 读
						lookupkey = DynamicDataSourceHolder.DB_SLAVE;
					}
				}
			}
		} else {
			// 有事务，表示非查询
			lookupkey = DynamicDataSourceHolder.DB_MASTET;
		}
		logger.debug("设置方法[{}] use[{}] Strategy,SqlCommanType[{}]..", ms.getId(), lookupkey,
				ms.getSqlCommandType().name());
		
		DynamicDataSourceHolder.setDbType(lookupkey);
		return invocation.proceed();
	}
	
	/**
	 * @desc 返回代理对象
	 */
	@Override
	public Object plugin(Object target) {
		// Executor,Plugin 是MyBatis 提供的类
		/*
		 * 当拦截的对象是 Executor 这个类型，就进行拦截，就将 intercept 包装到 wrap 中去。如果不是就直接返回本体，不受拦截
		 * Executor 在MyBatis 中是用来支持一系列增删改查操作。意识就是检测是有有CRUD操作，有就拦截，没有就放过
		 */
		if (target instanceof Executor) {
			return Plugin.wrap(target, this);
		} else {
			return target;
		}
	}

	@Override
	public void setProperties(Properties properties) {
	}
}
